package com.kk.config;

import com.kk.entity.Book;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Ds
 */
public class ReadExcel {
    // 总行数
    private int totalRows = 0;
    // 总条数
    private int totalCells = 0;
    // 错误信息接收器
    private String errorMsg;

    // 构造方法
    public ReadExcel() {
    }

    // 获取总行数
    public int getTotalRows() {
        return totalRows;
    }

    // 获取总列数
    public int getTotalCells() {
        return totalCells;
    }

    // 获取错误信息
    public String getErrorInfo() {
        return errorMsg;
    }

    /**
     * 读EXCEL文件，获取信息集合
     *
     * @param fielName
     * @return
     */
    public List<Book> getExcelInfo(MultipartFile mFile) {
        // 获取文件名
        String fileName = mFile.getOriginalFilename();

        List<Book> bookList = null;
        try {
            // 验证文件名是否合格
            if (!validateExcel(fileName)) {
                return null;
            }
            // 根据文件名判断文件是2003版本还是2007版本
            boolean isExcel2003 = true;
            if (isExcel2007(fileName)) {
                isExcel2003 = false;
            }
            bookList = createExcel(mFile.getInputStream(), isExcel2003);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return bookList;
    }

    /**
     * 根据excel里面的内容读取客户信息
     *
     * @param is输入流
     * @param isExcel2003 excel是2003还是2007版本
     * @return
     * @throws IOException
     */
    public List<Book> createExcel(InputStream is, boolean isExcel2003) {
        List<Book> bookList = null;
        try {
            Workbook wb = null;
            // 当excel是2003时,创建excel2003
            if (isExcel2003) {
                wb = new HSSFWorkbook(is);
            } else {// 当excel是2007时,创建excel2007
                wb = new XSSFWorkbook(is);
            }
            bookList = readExcelValue(wb);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return bookList;
    }

    /**
     * 读取Excel里面客户的信息
     *
     * @param wb
     * @return
     */
    private List<Book> readExcelValue(Workbook wb) {
        // 得到第一个shell
        Sheet sheet = wb.getSheetAt(0);
        // 得到Excel的行数
        this.totalRows = sheet.getPhysicalNumberOfRows();
        // 得到Excel的列数(前提是有行数)
        if (totalRows > 1 && sheet.getRow(0) != null) {
            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();
        }
        List<Book> userList = new ArrayList<Book>();
        // 循环Excel行数
        for (int r = 1; r < totalRows; r++) {
            Row row = (Row) sheet.getRow(r);
            if (row == null) {
                continue;
            }
            Book book = new Book();
            // 循环Excel的列
            for (int c = 0; c < this.totalCells; c++) {
                Cell cell = row.getCell(c);
                if (null != cell) {
                    switch(c)
                    {
                        case 0:book.setId(Integer.parseInt((cell.getStringCellValue())));
                            break;
                        case 1:book.setBookName(cell.getStringCellValue());
                            break;
                        case 2:book.setBookIcon(cell.getStringCellValue());
                            break;
                        case 3:book.setBookCode(cell.getStringCellValue());
                            break;
                        case 4:book.setBookCategory(cell.getStringCellValue());
                            break;
                        case 5:book.setBookVersion(cell.getStringCellValue());
                            break;
                        case 6:book.setBookGrade(cell.getStringCellValue());
                        break;
                        default:
                    }
//                  if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
//                            String age = String.valueOf(cell.getNumericCellValue());
//                            book.setAge(age.substring(0, age.length() - 2 > 0 ? age.length() - 2 : 1));
//                        } else {
//                            book.setAge(cell.getStringCellValue());
//                        }

                }
            }
            // 添加到list
            userList.add(book);
        }
        return userList;
    }

    /**
     * 验证EXCEL文件
     * @param filePath
     * @return
     */
    public boolean validateExcel(String filePath) {
        if (filePath == null || !(isExcel2003(filePath) || isExcel2007(filePath))) {
            errorMsg = "文件名不是excel格式";
            return false;
        }
        return true;
    }

    // @描述：是否是2003的excel，返回true是2003
    public static boolean isExcel2003(String filePath) {
        return filePath.matches("^.+\\.(?i)(xls)$");
    }

    // @描述：是否是2007的excel，返回true是2007
    public static boolean isExcel2007(String filePath) {
        return filePath.matches("^.+\\.(?i)(xlsx)$");
    }
}
